package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * 服务端类
 */
public class Server {
    private ExecutorService executorService = Executors.newFixedThreadPool(50);
    /**
     * ServerSocket: 服务端的主要组件
     * 1. 向系统申请服务端口，客户端通过该端口与服务端建立连接。
     * 2. 监听服务端口，一旦有客户端连接，就会立即创建一个Socket对象与客户端进行交互。
     * 如果将Socket比喻为“电话”，那ServerSocket就相当于客户中心的“总机”。
     */
    private ServerSocket serverSocket;

    // 使用Map存储所有客户端的输出流
    private Map<String, PrintWriter> allOut = new HashMap<>();

    public Server() {
        try {
            System.out.println("正在启动服务器");
            /**
             * 创建ServerSocket对象时，会申请一个端口。
             * 如果该端口被其他程序占用，会抛出异常：
             *   java.net.BindException: Address already in use:
             * 解决办法：
             * 1. 更换端口号。
             * 2. 杀死占用该端口的进程（通常由于服务端启动两次导致）。
             */
            serverSocket = new ServerSocket(8080);
            System.out.println("服务器启动完毕");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 启动服务器的方法
     */
    public void start() {
        try {
            while (true) {
                System.out.println("等待客户端连接...");
                /*
                 * accept(): 用于接收客户端连接，并返回一个Socket对象与所连接的客户端进行交互。
                 * 该方法是一个阻塞方法，调用后程序会“卡住”，直到一个客户端连接为止。
                 */
                Socket socket = serverSocket.accept();
                System.out.println("一个客户端连接了!");

                // 启动单独的线程来处理与客户端的通信
                Runnable handler = new ClientHandler(socket);
                executorService.execute(handler);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server server = new Server();
        server.start();
    }

    /**
     * 该线程任务用于负责与指定的客户端通信
     */
    private class ClientHandler implements Runnable {
        private Socket socket;
        private String host; // 记录客户端的IP地址
        private String nickname; // 记录客户端的昵称

        public ClientHandler(Socket socket) {
            this.socket = socket;
            // 通过socket获取该客户端的IP地址
            host = socket.getInetAddress().getHostAddress();
        }

        public void run() {
            PrintWriter pw = null;
            try {
                // 接收客户端发送过来的消息
                /*
                 * Socket提供了一个方法:
                 * InputStream getInputStream()
                 * 通过socket获取输入流，用于读取客户端发送过来的数据。
                 */
                InputStream in = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);
                // 读取客户端发送过来的第一行字符串一定是客户端的昵称
                nickname = br.readLine();

                /*
                 * 通过socket获取输出流，用于将消息发送给该客户端。
                 */
                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out, StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw, true);

                // 将对应客户端的输出流存入到共享集合allOut中
                synchronized (allOut) { // 同步监视器对象选取临界资源（多个线程要操作同一个集合，锁该集合）
                    allOut.put(nickname, pw);
                }
                // 广播该客户端上线了
                // 范传奇[192.168.1.124]上线了,当前在线人数:2
                Set<String> keys = allOut.keySet();
                broadcast(nickname + "[" + host + "]上线了，当前在线人数:" + allOut.size() + "\n" + "当前在线用户:" + keys);

                /*
                 * 当我们通过br.readLine()读取客户端发送过来的数据时，
                 * 客户端如果调用socket.close()时（进行四次挥手），正常与我们断开连接，
                 * 那么readLine()会返回null；
                 * 若客户端异常中断（强制杀死进程），此时readLine()会抛出异常。
                 * 注意：这种情况咱们服务端是没有办法的。
                 */
                String message;
                while ((message = br.readLine()) != null) { // 读取客户端发送过来的消息
                    if (message.startsWith("@")) {
                        // 如果消息是以"@"开头则为私聊功能
                        sendMessageToSomeone(message);

                    } else {
                        broadcast(nickname + "[" + host + "]说:" + message);
                        saveMessage(nickname, null,message);
                    }
                }
            } catch (IOException e) {
                // 处理客户端断开连接的情况
            } finally {
                // 处理该客户端断开链接后的操作
                // 将该客户端的输出流从共享集合allOut中删除
                synchronized (allOut) {
                    allOut.remove(nickname);
                }
                // 广播消息，告知所有客户端该用户下线了
                Set<String> keys = allOut.keySet();
                broadcast(nickname + "[" + host + "]下线了，当前在线人数:" + allOut.size() + "\n" + "当前在线用户:" + keys);

                try {
                    socket.close(); // 进行四次挥手
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        /**
         * 将消息发送给所有客户端
         *
         * @param message 消息内容
         */
        private void broadcast(String message) {
            System.out.println(message);
            synchronized (allOut) {
                for (PrintWriter o : allOut.values()) {
                    o.println(message);
                }
            }
        }

        private void sendMessageToSomeone(String message) {
            if (message.matches("@.+:.+")) {
                String toNickName = message.substring(1, message.indexOf(":"));
                if (allOut.containsKey(toNickName)) {
                    PrintWriter pw = allOut.get(toNickName);
                    String toMessage = message.substring(message.indexOf(":"));
                    pw.println(nickname + "和" + toNickName + "的二人世界,发生了" + toMessage);
                    saveMessage(nickname, toNickName, toMessage);
                } else {
                    PrintWriter pw = allOut.get(nickname);
                    pw.println("不存在该用户");
                }
            } else {
                PrintWriter pw = allOut.get(nickname);
                pw.println("创建二人世界格式不正确 正确格式为 @对方昵称:内容");
            }
        }

        private void saveMessage(String fromUser, String toUser, String message) {
            try (
                    Connection conn = DBUtil.getConnection();
            ) {
                String sql = "INSERT INTO chatinfo (from_user, to_user, content) VALUES (?, ?, ?)";
                PreparedStatement ps = conn.prepareStatement(sql);
                ps.setString(1, fromUser);
                ps.setString(2, toUser);
                ps.setString(3, message);
                int num = ps.executeUpdate();
                if (num > 0) {
                    System.out.println("记录已保存");
                }
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }
}
